home *** CD-ROM | disk | FTP | other *** search
/ CD Actual 2 / CD ACTUAL VOL 2.iso / install / fips / restorrb / restorrb.c < prev    next >
Encoding:
C/C++ Source or Header  |  1995-09-10  |  6.3 KB  |  226 lines
  1. /* 
  2.     FIPS - the First nondestructive Interactive Partition Splitting program 
  3.     Module restorrb.c 
  4.  
  5.     Copyright (C) 1993 Arno Schaefer 
  6.  
  7.     This program is free software; you can redistribute it and/or modify 
  8.     it under the terms of the GNU General Public License as published by 
  9.     the Free Software Foundation; either version 1, or (at your option) 
  10.     any later version. 
  11.  
  12.     This program is distributed in the hope that it will be useful, 
  13.     but WITHOUT ANY WARRANTY; without even the implied warranty of 
  14.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
  15.     GNU General Public License for more details. 
  16.  
  17.     You should have received a copy of the GNU General Public License 
  18.     along with this program; if not, write to the Free Software 
  19.     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. 
  20. */ 
  21.  
  22. #include <stdio.h> 
  23. #include <io.h> 
  24. #include <stdlib.h> 
  25. #include <dos.h> 
  26. #include <bios.h> 
  27. #include <alloc.h> 
  28. #include <conio.h> 
  29. #include <ctype.h> 
  30.  
  31. #include "rtypes.h" 
  32.  
  33. #define DISK_INT 0x13 
  34.  
  35. #define RESET_DISK 0 
  36. #define WRITE_SECTOR 3 
  37. #define VERIFY_SECTOR 4 
  38.  
  39. #define DISK1 0x80 
  40.  
  41. /* ----------------------------------------------------------------------- */ 
  42. /* Copyright notice and version number                                     */ 
  43. /* ----------------------------------------------------------------------- */ 
  44.  
  45. void notice (void) 
  46. { 
  47.     printf ("\nFIPS version 1.1, Copyright (C) 1993/94 Arno Schaefer\n"); 
  48.     printf ("Module RESTORRB.EXE - Please read the file README.1ST\n"); 
  49.     printf ("FIPS comes with ABSOLUTELY NO WARRANTY, see file COPYING for details\n"); 
  50.     printf ("This is free software, and you are welcome to redistribute it\n"); 
  51.     printf ("under certain conditions; again see file COPYING for details.\n"); 
  52. } 
  53.  
  54. /* ----------------------------------------------------------------------- */ 
  55. /* Error Handling                                                          */ 
  56. /* ----------------------------------------------------------------------- */ 
  57.  
  58. int getx (void) 
  59. { 
  60.     int character = getch(); 
  61.  
  62.     if (character == 3) 
  63.     { 
  64.         printf ("\n"); 
  65.         exit (0); 
  66.     } 
  67.     return (character); 
  68. } 
  69.  
  70. void error (char *message) 
  71. { 
  72.     fprintf (stderr,"\nError: %s!\n",message); 
  73.     exit (-1); 
  74. } 
  75.  
  76. /* ----------------------------------------------------------------------- */ 
  77. /* BIOS calls                                                              */ 
  78. /* ----------------------------------------------------------------------- */ 
  79.  
  80. int reset_drives (void) 
  81. { 
  82.     union REGS regs; 
  83.  
  84.     regs.h.ah = RESET_DISK; 
  85.     regs.h.dl = DISK1; 
  86.     int86 (DISK_INT,®s,®s); 
  87.     if (regs.x.cflag) return (-1); 
  88.     return 0; 
  89. } 
  90.  
  91. /* ----------------------------------------------------------------------- */ 
  92. /* read / write sectors                                                    */ 
  93. /* ----------------------------------------------------------------------- */ 
  94.  
  95. int verify_sector (int drive_number,dword head,dword cylinder,dword sector,byte *buffer) 
  96. { 
  97.     if (biosdisk (VERIFY_SECTOR,drive_number,head,cylinder,sector,1,buffer)) return (-1); 
  98.     return 0; 
  99. } 
  100.  
  101. int write_sector (int drive_number,dword head,dword cylinder,dword sector,byte *buffer) 
  102. { 
  103.     int i; 
  104.     boolean done=false; 
  105.     for (i=0;i<3;i++) 
  106.     { 
  107.         if (!biosdisk (WRITE_SECTOR,drive_number,head,cylinder,sector,1,buffer)) 
  108.         { 
  109.             done=true; 
  110.             break; 
  111.         } 
  112.         reset_drives(); 
  113.     } 
  114.     if (!done) return (-1); 
  115.     return (verify_sector (drive_number,head,cylinder,sector,buffer)); 
  116. } 
  117.  
  118. int write_root_sector (int drive_number,byte *buffer) 
  119. { 
  120.     return (write_sector (drive_number,0,0,1,buffer)); 
  121. } 
  122.  
  123. /* ----------------------------------------------------------------------- */ 
  124. /* User Input                                                              */ 
  125. /* ----------------------------------------------------------------------- */ 
  126.  
  127. void ask_for_write_permission (void) 
  128. { 
  129.     int character = 'x'; 
  130.  
  131.     printf ("\nReady to write old root- and bootsector to disk\n"); 
  132.     printf ("Do you want to proceed (y/n): "); 
  133.  
  134.     while ((character != 'y') && (character != 'n')) character = getx(); 
  135.     printf ("%c\n",character); 
  136.     if (character == 'n') exit (0); 
  137. } 
  138.  
  139. /* ----------------------------------------------------------------------- */ 
  140. /* Main                                                                    */ 
  141. /* ----------------------------------------------------------------------- */ 
  142.  
  143. void main (void) 
  144. { 
  145.     byte rootsector[512]; 
  146.     byte bootsector[512]; 
  147.     int drive_number,partition_number,i; 
  148.     FILE *handle; 
  149.     dword head,cylinder,sector; 
  150.     char *filename = "a:\\rootboot.000"; 
  151.     int no_of_savefiles = 0; 
  152.     char first = 'x'; 
  153.  
  154.     notice(); 
  155.  
  156.     if (reset_drives ()) error ("Drive Initialization Failure"); 
  157.  
  158.     for (i='0';i<='9';i++) 
  159.     { 
  160.         filename[14] = i; 
  161.         if (access (filename,0) == 0) 
  162.         { 
  163.             if (first == 'x') first = i; 
  164.             no_of_savefiles++; 
  165.             printf ("found savefile %s\n",filename); 
  166.         } 
  167.     } 
  168.  
  169.     if (no_of_savefiles == 0) error ("No savefile ROOTBOOT.00? found on disk A:"); 
  170.  
  171.     if (no_of_savefiles > 1) 
  172.     { 
  173.         printf ("\nWhich file do you want to restore? "); 
  174.         while (true) 
  175.         { 
  176.             int c; 
  177.             if (isdigit (c = getx())) 
  178.             { 
  179.                 filename[14] = c; 
  180.                 if (access (filename,0) == 0) break; 
  181.             } 
  182.         } 
  183.     } 
  184.     else 
  185.     { 
  186.         filename[14] = first; 
  187.     } 
  188.  
  189.     if (no_of_savefiles > 1) 
  190.     { 
  191.         printf ("%c\n", filename[14]); 
  192.     } 
  193.  
  194.     if ((handle = fopen (filename,"rb")) == NULL) 
  195.         error ("Can't open file"); 
  196.  
  197.     for (i=0;i<512;i++) 
  198.     { 
  199.         int character = fgetc (handle); 
  200.         if (character == EOF) error ("Error reading file from disk"); 
  201.         *(rootsector + i) = character; 
  202.     } 
  203.     for (i=0;i<512;i++) 
  204.     { 
  205.         int character = fgetc (handle); 
  206.         if (character == EOF) error ("Error reading file from disk"); 
  207.         *(bootsector + i) = character; 
  208.     } 
  209.     if ((drive_number = fgetc (handle)) == EOF) error ("Error reading file from disk"); 
  210.     if ((partition_number = fgetc (handle)) == EOF) error ("Error reading file from disk"); 
  211.     if (fclose (handle)) error ("Error closing file"); 
  212.  
  213.     head = (dword) rootsector[0x1be+16*partition_number+1]; 
  214.     cylinder = (((dword) rootsector[0x1be+16*partition_number+2] << 2) & 0x300) 
  215.         | (dword) rootsector[0x1be+16*partition_number+3]; 
  216.     sector = (dword) rootsector[0x1be+16*partition_number+2] & 0x3f; 
  217.  
  218.     ask_for_write_permission(); 
  219.  
  220.     if (write_root_sector (drive_number,rootsector)) 
  221.         error ("Error writing rootsector"); 
  222.  
  223.     if (write_sector (drive_number,head,cylinder,sector,bootsector)) 
  224.         error ("Error writing bootsector"); 
  225. } 
  226.